package com.qianli.cashmama.trade.mq;

import com.aliyun.openservices.ons.api.*;
import com.fqgj.common.utils.StringUtils;
import com.qianli.cashmama.trade.common.mq.MessageMQ;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.aop.framework.AopProxyUtils;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

/**
 * @program: market-api
 * @description
 * @author: frank
 * @create: 2020-03-23 14:58
 **/
@Component
public class MessageMQHandler implements ApplicationContextAware {
    private static final Logger LOGGER = LoggerFactory.getLogger(MessageMQHandler.class);

    @Autowired
    private RocketMQProperties rocketMQProperties;

    private Producer producer;

    /**
     * 发送消息
     *
     * @param message
     */
    public void send(MessageMQ message) {
        Message msg = new Message( //
                // Message 所属的 Topic
                message.getTopic().getName(),
                // Message Tag 可理解为 Gmail 中的标签，对消息进行再归类，方便 Consumer 指定过滤条件在 MQ 服务器过滤
                message.getTag().getDesc(),
                // Message Body 可以是任何二进制形式的数据， MQ 不做任何干预，
                // 需要 Producer 与 Consumer 协商好一致的序列化和反序列化方式
                message.getMessage().getBytes());
        // 设置代表消息的业务关键属性，请尽可能全局唯一。
        // 以方便您在无法正常收到消息情况下，可通过阿里云服务器管理控制台查询消息并补发
        // 注意：不设置也不会影响消息正常收发
        msg.setKey(message.getMessageId());

        try {
            SendResult sendResult = producer.send(msg);
            // 同步发送消息，只要不抛异常就是成功
            if (sendResult != null) {
                LOGGER.info("Send mq message success. Topic is:{}, msgId is: {}", message.getTopic(), sendResult.getMessageId());
            }
        } catch (Exception e) {
            // 消息发送失败，需要进行重试处理，可重新发送这条消息或持久化这条数据进行补偿处理
            LOGGER.error("Send mq message fail. Topic is:{}, e : {}", message.getTopic(), e);
        }
    }

    //rm-bp17d31bej5lb9fp2lo.mysql.rds.aliyuncs.com

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        if (StringUtils.isEmpty(rocketMQProperties.getGroup())) {
            LOGGER.info("未配置mq地址信息 不启动mq");
            return;
        }
        Map<String, Object> beansWithAnnotation = applicationContext.getBeansWithAnnotation(RocketMQTag.class);
        Map<String, RocketMQConsumerHelper> rocketMQConsumerHelperMap = new HashMap<>(10);
        beansWithAnnotation.forEach((name, bean) -> {
            Class<?> clazz = AopProxyUtils.ultimateTargetClass(bean);
            RocketMQTag rocketMqTag = clazz.getAnnotation(RocketMQTag.class);
            rocketMQConsumerHelperMap.put(rocketMqTag.value(), (RocketMQConsumerHelper) bean);
        });

        Properties properties = new Properties();
        properties.setProperty(PropertyKeyConst.GROUP_ID, rocketMQProperties.getGroup());
        // AccessKey 阿里云身份验证，在阿里云服务器管理控制台创建
        properties.put(PropertyKeyConst.AccessKey, rocketMQProperties.getAccessKey());
        // SecretKey 阿里云身份验证，在阿里云服务器管理控制台创建
        properties.put(PropertyKeyConst.SecretKey, rocketMQProperties.getSecretKey());
        //设置发送超时时间，单位毫秒
        properties.setProperty(PropertyKeyConst.SendMsgTimeoutMillis, "3000");
        // 设置 TCP 接入域名，到控制台的实例基本信息中查看
        properties.put(PropertyKeyConst.NAMESRV_ADDR, rocketMQProperties.getNameServer());

        producer = ONSFactory.createProducer(properties);
        // 在发送消息前，必须调用 start 方法来启动 Producer，只需调用一次即可
        producer.start();

//        Consumer consumer = ONSFactory.createConsumer(properties);
//        //订阅多个 Tag
//        consumer.subscribe(MQTopicEnum.OPEN_API.getName(), "*", (message, context) -> {
//            LOGGER.info("mq message receive:{}", message);
//            try {
//                rocketMQConsumerHelperMap.get(message.getTag()).onMessage(new String(message.getBody()));
//            } catch (Exception e) {
//                e.printStackTrace();
//                return Action.ReconsumeLater;
//            }
//            return Action.CommitMessage;
//        });
//        consumer.start();
    }
}
